package com.zang.tree;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * 二叉搜索树中的众数
 * @author ZVerify
 * @since 2022/11/04 14:48
 * @see <a href="https://leetcode.cn/problems/find-mode-in-binary-search-tree/description/">...</a>
 **/
public class 二叉搜索树中的众数 {

    // 存放结果集
    List<Integer> list = new ArrayList<>();
    // 双指针的上一节点
    TreeNode pre = null;
    // 当前节点的出现数量
    int count = 0;
    // 最大值个数
    int maxCount = 0;
    public int[] findMode(TreeNode root) {
        dfs(root);
        return list.stream().mapToInt(a->a).distinct().toArray();
    }

    private void dfs(TreeNode root) {
        if (root==null) return;
        dfs(root.left);
        // 中序遍历此时处理节点
        // 判断是否是初始节点或者一个新的值
        if (pre==null || pre.val!=root.val){
            count = 1;
        }else {
            count++;
        }
        // 如果总数大于了最大出现次数则更新结果集并且更新最大值
        if (count>maxCount) {
            list.clear();
            list.add(root.val);
            maxCount=count;
        }
        // 可能不是一个众数可能包含多个
        if (count==maxCount){
            list.add(root.val);
        }
        // 更新当前节点为前置节点
        pre = root;
        dfs(root.right);
    }
}
